from __future__ import annotations

from datetime import timedelta
from typing import Any
from unittest import mock

import pytest
from django.urls import reverse

from sentry.discover.models import DatasetSourcesTypes
from sentry.models.dashboard_widget import DashboardWidget, DashboardWidgetTypes
from sentry.models.environment import Environment
from sentry.sentry_metrics.use_case_id_registry import UseCaseID
from sentry.snuba.metrics.extraction import MetricSpecType, OnDemandMetricSpec
from sentry.testutils.cases import MetricsEnhancedPerformanceTestCase
from sentry.testutils.helpers.datetime import before_now
from sentry.testutils.helpers.on_demand import create_widget
from sentry.utils.samples import load_data

pytestmark = pytest.mark.sentry_metrics


class OrganizationEventsStatsMetricsEnhancedPerformanceEndpointTest(
    MetricsEnhancedPerformanceTestCase
):
    endpoint = "sentry-api-0-organization-events-stats"
    METRIC_STRINGS = [
        "foo_transaction",
        "d:transactions/measurements.datacenter_memory@pebibyte",
    ]

    def setUp(self) -> None:
        super().setUp()
        self.login_as(user=self.user)
        self.day_ago = before_now(days=1).replace(hour=10, minute=0, second=0, microsecond=0)
        self.DEFAULT_METRIC_TIMESTAMP = self.day_ago

        self.url = reverse(
            "sentry-api-0-organization-events-stats",
            kwargs={"organization_id_or_slug": self.project.organization.slug},
        )
        self.features = {
            "organizations:performance-use-metrics": True,
        }

        self.additional_params: dict[str, Any] = dict()

    # These throughput tests should roughly match the ones in OrganizationEventsStatsEndpointTest
    @pytest.mark.querybuilder
    def test_throughput_epm_hour_rollup(self) -> None:
        # Each of these denotes how many events to create in each hour
        event_counts = [6, 0, 6, 3, 0, 3]
        for hour, count in enumerate(event_counts):
            for minute in range(count):
                self.store_transaction_metric(
                    1, timestamp=self.day_ago + timedelta(hours=hour, minutes=minute)
                )

        for axis in ["epm()", "tpm()"]:
            response = self.do_request(
                data={
                    "start": self.day_ago,
                    "end": self.day_ago + timedelta(hours=6),
                    "interval": "1h",
                    "yAxis": axis,
                    "project": self.project.id,
                    "dataset": "metricsEnhanced",
                    **self.additional_params,
                },
            )
            assert response.status_code == 200, response.content
            data = response.data["data"]
            assert len(data) == 6
            assert response.data["isMetricsData"]

            rows = data[0:6]
            for test in zip(event_counts, rows):
                assert test[1][1][0]["count"] == test[0] / (3600.0 / 60.0)

    @pytest.mark.querybuilder
    def test_throughput_spm_hour_rollup(self) -> None:
        # Each of these denotes how many events to create in each hour
        event_counts = [6, 0, 6, 3, 0, 3]
        for hour, count in enumerate(event_counts):
            for minute in range(count):
                self.store_span_metric(
                    1,
                    timestamp=self.day_ago + timedelta(hours=hour, minutes=minute),
                )

        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=6),
                "interval": "1h",
                "yAxis": "spm()",
                "project": self.project.id,
                "dataset": "metrics",
                **self.additional_params,
            },
        )

        assert response.status_code == 200, response.content
        data = response.data["data"]
        assert len(data) == 6
        assert response.data["meta"]["dataset"] == "metrics"

        rows = data[0:6]

        for test in zip(event_counts, rows):
            assert test[1][1][0]["count"] == test[0] / (3600.0 / 60.0)

    def test_throughput_epm_day_rollup(self) -> None:
        # Each of these denotes how many events to create in each minute
        event_counts = [6, 0, 6, 3, 0, 3]
        for hour, count in enumerate(event_counts):
            for minute in range(count):
                self.store_transaction_metric(
                    1, timestamp=self.day_ago + timedelta(hours=hour, minutes=minute)
                )

        for axis in ["epm()", "tpm()"]:
            response = self.do_request(
                data={
                    "start": self.day_ago,
                    "end": self.day_ago + timedelta(hours=24),
                    "interval": "24h",
                    "yAxis": axis,
                    "project": self.project.id,
                    "dataset": "metricsEnhanced",
                    **self.additional_params,
                },
            )
            assert response.status_code == 200, response.content
            data = response.data["data"]
            assert len(data) == 2
            assert response.data["isMetricsData"]

            assert data[0][1][0]["count"] == sum(event_counts) / (86400.0 / 60.0)

    def test_throughput_epm_hour_rollup_offset_of_hour(self) -> None:
        # Each of these denotes how many events to create in each hour
        event_counts = [6, 0, 6, 3, 0, 3]
        for hour, count in enumerate(event_counts):
            for minute in range(count):
                self.store_transaction_metric(
                    1, timestamp=self.day_ago + timedelta(hours=hour, minutes=minute + 30)
                )

        for axis in ["tpm()", "epm()"]:
            response = self.do_request(
                data={
                    "start": self.day_ago + timedelta(minutes=30),
                    "end": self.day_ago + timedelta(hours=6, minutes=30),
                    "interval": "1h",
                    "yAxis": axis,
                    "project": self.project.id,
                    "dataset": "metricsEnhanced",
                    **self.additional_params,
                },
            )
            assert response.status_code == 200, response.content
            data = response.data["data"]
            assert len(data) == 6
            assert response.data["isMetricsData"]

            rows = data[0:6]
            for test in zip(event_counts, rows):
                assert test[1][1][0]["count"] == test[0] / (3600.0 / 60.0)

    def test_throughput_eps_minute_rollup(self) -> None:
        # Each of these denotes how many events to create in each minute
        event_counts = [6, 0, 6, 3, 0, 3]
        for minute, count in enumerate(event_counts):
            for second in range(count):
                self.store_transaction_metric(
                    1, timestamp=self.day_ago + timedelta(minutes=minute, seconds=second)
                )

        for axis in ["eps()", "tps()"]:
            response = self.do_request(
                data={
                    "start": self.day_ago,
                    "end": self.day_ago + timedelta(minutes=6),
                    "interval": "1m",
                    "yAxis": axis,
                    "project": self.project.id,
                    "dataset": "metricsEnhanced",
                    **self.additional_params,
                },
            )
            assert response.status_code == 200, response.content
            data = response.data["data"]
            assert len(data) == 6
            assert response.data["isMetricsData"]

            rows = data[0:6]
            for test in zip(event_counts, rows):
                assert test[1][1][0]["count"] == test[0] / 60.0

    def test_failure_rate(self) -> None:
        for hour in range(6):
            timestamp = self.day_ago + timedelta(hours=hour, minutes=30)
            self.store_transaction_metric(1, tags={"transaction.status": "ok"}, timestamp=timestamp)
            if hour < 3:
                self.store_transaction_metric(
                    1, tags={"transaction.status": "internal_error"}, timestamp=timestamp
                )

        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=6),
                "interval": "1h",
                "yAxis": ["failure_rate()"],
                "project": self.project.id,
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        data = response.data["data"]
        assert len(data) == 6
        assert response.data["isMetricsData"]
        assert [attrs for time, attrs in response.data["data"]] == [
            [{"count": 0.5}],
            [{"count": 0.5}],
            [{"count": 0.5}],
            [{"count": 0}],
            [{"count": 0}],
            [{"count": 0}],
        ]

    def test_percentiles_multi_axis(self) -> None:
        for hour in range(6):
            timestamp = self.day_ago + timedelta(hours=hour, minutes=30)
            self.store_transaction_metric(111, timestamp=timestamp)
            self.store_transaction_metric(222, metric="measurements.lcp", timestamp=timestamp)

        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=6),
                "interval": "1h",
                "yAxis": ["p75(measurements.lcp)", "p75(transaction.duration)"],
                "project": self.project.id,
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        lcp = response.data["p75(measurements.lcp)"]
        duration = response.data["p75(transaction.duration)"]
        assert len(duration["data"]) == 6
        assert duration["isMetricsData"]
        assert len(lcp["data"]) == 6
        assert lcp["isMetricsData"]
        for item in duration["data"]:
            assert item[1][0]["count"] == 111
        for item in lcp["data"]:
            assert item[1][0]["count"] == 222

    @mock.patch("sentry.snuba.metrics_enhanced_performance.timeseries_query", return_value={})
    def test_multiple_yaxis_only_one_query(self, mock_query: mock.MagicMock) -> None:
        self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": ["epm()", "eps()", "tpm()", "p50(transaction.duration)"],
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )

        assert mock_query.call_count == 1

    def test_aggregate_function_user_count(self) -> None:
        self.store_transaction_metric(
            1, metric="user", timestamp=self.day_ago + timedelta(minutes=30)
        )
        self.store_transaction_metric(
            1, metric="user", timestamp=self.day_ago + timedelta(hours=1, minutes=30)
        )
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": "count_unique(user)",
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        assert response.data["isMetricsData"]
        assert [attrs for time, attrs in response.data["data"]] == [[{"count": 1}], [{"count": 1}]]
        meta = response.data["meta"]
        assert meta["isMetricsData"] == response.data["isMetricsData"]

    def test_non_mep_query_fallsback(self) -> None:
        def get_mep(query):
            response = self.do_request(
                data={
                    "project": self.project.id,
                    "start": self.day_ago,
                    "end": self.day_ago + timedelta(hours=2),
                    "interval": "1h",
                    "query": query,
                    "yAxis": ["epm()"],
                    "dataset": "metricsEnhanced",
                    **self.additional_params,
                },
            )
            assert response.status_code == 200, response.content
            return response.data["isMetricsData"]

        assert get_mep(""), "empty query"
        assert get_mep("event.type:transaction"), "event type transaction"
        assert not get_mep("event.type:error"), "event type error"
        assert not get_mep("transaction.duration:<15min"), "outlier filter"
        assert get_mep("epm():>0.01"), "throughput filter"
        assert not get_mep(
            "event.type:transaction OR event.type:error"
        ), "boolean with non-mep filter"
        assert get_mep(
            "event.type:transaction OR transaction:foo_transaction"
        ), "boolean with mep filter"

    def test_having_condition_with_preventing_aggregates(self) -> None:
        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "query": "p95():<5s",
                "yAxis": ["epm()"],
                "dataset": "metricsEnhanced",
                "preventMetricAggregates": "1",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        assert not response.data["isMetricsData"]
        meta = response.data["meta"]
        assert meta["isMetricsData"] == response.data["isMetricsData"]

    def test_explicit_not_mep(self) -> None:
        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                # Should be a mep able query
                "query": "",
                "yAxis": ["epm()"],
                "metricsEnhanced": "0",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        assert not response.data["isMetricsData"]
        meta = response.data["meta"]
        assert meta["isMetricsData"] == response.data["isMetricsData"]

    def test_sum_transaction_duration(self) -> None:
        self.store_transaction_metric(123, timestamp=self.day_ago + timedelta(minutes=30))
        self.store_transaction_metric(456, timestamp=self.day_ago + timedelta(hours=1, minutes=30))
        self.store_transaction_metric(789, timestamp=self.day_ago + timedelta(hours=1, minutes=30))
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": "sum(transaction.duration)",
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        assert response.data["isMetricsData"]
        assert [attrs for time, attrs in response.data["data"]] == [
            [{"count": 123}],
            [{"count": 1245}],
        ]
        meta = response.data["meta"]
        assert meta["isMetricsData"] == response.data["isMetricsData"]
        assert meta["fields"] == {"time": "date", "sum_transaction_duration": "duration"}
        assert meta["units"] == {"time": None, "sum_transaction_duration": "millisecond"}

    def test_sum_transaction_duration_with_comparison(self) -> None:
        # We store the data for the previous day (in order to have values for the comparison).
        self.store_transaction_metric(
            1, timestamp=self.day_ago - timedelta(days=1) + timedelta(minutes=30)
        )
        self.store_transaction_metric(
            2, timestamp=self.day_ago - timedelta(days=1) + timedelta(minutes=30)
        )
        # We store the data for today.
        self.store_transaction_metric(123, timestamp=self.day_ago + timedelta(minutes=30))
        self.store_transaction_metric(456, timestamp=self.day_ago + timedelta(minutes=30))
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(days=1),
                "interval": "1d",
                "yAxis": "sum(transaction.duration)",
                "comparisonDelta": 86400,
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        assert response.data["isMetricsData"]
        # For some reason, if all tests run, there is some shared state that makes this test have data in the second
        # time bucket, which is filled automatically by the zerofilling. In order to avoid this flaky failure, we will
        # only check that the first bucket contains the actual data.
        assert [attrs for time, attrs in response.data["data"]][0] == [
            {"comparisonCount": 3.0, "count": 579.0}
        ]
        meta = response.data["meta"]
        assert meta["isMetricsData"] == response.data["isMetricsData"]
        assert meta["fields"] == {"time": "date", "sum_transaction_duration": "duration"}
        assert meta["units"] == {"time": None, "sum_transaction_duration": "millisecond"}

    def test_custom_measurement(self) -> None:
        self.store_transaction_metric(
            123,
            metric="measurements.bytes_transfered",
            internal_metric="d:transactions/measurements.datacenter_memory@pebibyte",
            entity="metrics_distributions",
            tags={"transaction": "foo_transaction"},
            timestamp=self.day_ago + timedelta(minutes=30),
        )
        self.store_transaction_metric(
            456,
            metric="measurements.bytes_transfered",
            internal_metric="d:transactions/measurements.datacenter_memory@pebibyte",
            entity="metrics_distributions",
            tags={"transaction": "foo_transaction"},
            timestamp=self.day_ago + timedelta(hours=1, minutes=30),
        )
        self.store_transaction_metric(
            789,
            metric="measurements.bytes_transfered",
            internal_metric="d:transactions/measurements.datacenter_memory@pebibyte",
            entity="metrics_distributions",
            tags={"transaction": "foo_transaction"},
            timestamp=self.day_ago + timedelta(hours=1, minutes=30),
        )
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": "sum(measurements.datacenter_memory)",
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        assert response.data["isMetricsData"]
        assert [attrs for time, attrs in response.data["data"]] == [
            [{"count": 123}],
            [{"count": 1245}],
        ]
        meta = response.data["meta"]
        assert meta["isMetricsData"] == response.data["isMetricsData"]
        assert meta["fields"] == {"time": "date", "sum_measurements_datacenter_memory": "size"}
        assert meta["units"] == {"time": None, "sum_measurements_datacenter_memory": "pebibyte"}

    def test_does_not_fallback_if_custom_metric_is_out_of_request_time_range(self) -> None:
        self.store_transaction_metric(
            123,
            timestamp=self.day_ago + timedelta(hours=1),
            internal_metric="d:transactions/measurements.custom@kibibyte",
            entity="metrics_distributions",
        )
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": "p99(measurements.custom)",
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        meta = response.data["meta"]
        assert response.status_code == 200, response.content
        assert response.data["isMetricsData"]
        assert meta["isMetricsData"]
        assert meta["fields"] == {"time": "date", "p99_measurements_custom": "size"}
        assert meta["units"] == {"time": None, "p99_measurements_custom": "kibibyte"}

    def test_multi_yaxis_custom_measurement(self) -> None:
        self.store_transaction_metric(
            123,
            metric="measurements.bytes_transfered",
            internal_metric="d:transactions/measurements.datacenter_memory@pebibyte",
            entity="metrics_distributions",
            tags={"transaction": "foo_transaction"},
            timestamp=self.day_ago + timedelta(minutes=30),
        )
        self.store_transaction_metric(
            456,
            metric="measurements.bytes_transfered",
            internal_metric="d:transactions/measurements.datacenter_memory@pebibyte",
            entity="metrics_distributions",
            tags={"transaction": "foo_transaction"},
            timestamp=self.day_ago + timedelta(hours=1, minutes=30),
        )
        self.store_transaction_metric(
            789,
            metric="measurements.bytes_transfered",
            internal_metric="d:transactions/measurements.datacenter_memory@pebibyte",
            entity="metrics_distributions",
            tags={"transaction": "foo_transaction"},
            timestamp=self.day_ago + timedelta(hours=1, minutes=30),
        )
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": [
                    "sum(measurements.datacenter_memory)",
                    "p50(measurements.datacenter_memory)",
                ],
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        sum_data = response.data["sum(measurements.datacenter_memory)"]
        p50_data = response.data["p50(measurements.datacenter_memory)"]
        assert sum_data["isMetricsData"]
        assert p50_data["isMetricsData"]
        assert [attrs for time, attrs in sum_data["data"]] == [
            [{"count": 123}],
            [{"count": 1245}],
        ]
        assert [attrs for time, attrs in p50_data["data"]] == [
            [{"count": 123}],
            [{"count": 622.5}],
        ]

        sum_meta = sum_data["meta"]
        assert sum_meta["isMetricsData"] == sum_data["isMetricsData"]
        assert sum_meta["fields"] == {
            "time": "date",
            "sum_measurements_datacenter_memory": "size",
            "p50_measurements_datacenter_memory": "size",
        }
        assert sum_meta["units"] == {
            "time": None,
            "sum_measurements_datacenter_memory": "pebibyte",
            "p50_measurements_datacenter_memory": "pebibyte",
        }

        p50_meta = p50_data["meta"]
        assert p50_meta["isMetricsData"] == p50_data["isMetricsData"]
        assert p50_meta["fields"] == {
            "time": "date",
            "sum_measurements_datacenter_memory": "size",
            "p50_measurements_datacenter_memory": "size",
        }
        assert p50_meta["units"] == {
            "time": None,
            "sum_measurements_datacenter_memory": "pebibyte",
            "p50_measurements_datacenter_memory": "pebibyte",
        }

    def test_dataset_metrics_does_not_fallback(self) -> None:
        self.store_transaction_metric(123, timestamp=self.day_ago + timedelta(minutes=30))
        self.store_transaction_metric(456, timestamp=self.day_ago + timedelta(hours=1, minutes=30))
        self.store_transaction_metric(789, timestamp=self.day_ago + timedelta(hours=1, minutes=30))
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "query": "transaction.duration:<5s",
                "yAxis": "sum(transaction.duration)",
                "dataset": "metrics",
                **self.additional_params,
            },
        )
        assert response.status_code == 400, response.content

    def test_title_filter(self) -> None:
        self.store_transaction_metric(
            123,
            tags={"transaction": "foo_transaction"},
            timestamp=self.day_ago + timedelta(minutes=30),
        )
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "query": "title:foo_transaction",
                "yAxis": [
                    "sum(transaction.duration)",
                ],
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        data = response.data["data"]
        assert [attrs for time, attrs in data] == [
            [{"count": 123}],
            [{"count": 0}],
        ]

    def test_transaction_status_unknown_error(self) -> None:
        self.store_transaction_metric(
            123,
            tags={"transaction.status": "unknown"},
            timestamp=self.day_ago + timedelta(minutes=30),
        )
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "query": "transaction.status:unknown_error",
                "yAxis": [
                    "sum(transaction.duration)",
                ],
                "dataset": "metricsEnhanced",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        data = response.data["data"]
        assert [attrs for time, attrs in data] == [
            [{"count": 123}],
            [{"count": 0}],
        ]

    def test_custom_performance_metric_meta_contains_field_and_unit_data(self) -> None:
        self.store_transaction_metric(
            123,
            timestamp=self.day_ago + timedelta(hours=1),
            internal_metric="d:transactions/measurements.custom@kibibyte",
            entity="metrics_distributions",
        )
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": "p99(measurements.custom)",
                "query": "",
                **self.additional_params,
            },
        )

        assert response.status_code == 200
        meta = response.data["meta"]
        assert meta["fields"] == {"time": "date", "p99_measurements_custom": "size"}
        assert meta["units"] == {"time": None, "p99_measurements_custom": "kibibyte"}

    def test_multi_series_custom_performance_metric_meta_contains_field_and_unit_data(self) -> None:
        self.store_transaction_metric(
            123,
            timestamp=self.day_ago + timedelta(hours=1),
            internal_metric="d:transactions/measurements.custom@kibibyte",
            entity="metrics_distributions",
        )
        self.store_transaction_metric(
            123,
            timestamp=self.day_ago + timedelta(hours=1),
            internal_metric="d:transactions/measurements.another.custom@pebibyte",
            entity="metrics_distributions",
        )
        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": [
                    "p95(measurements.custom)",
                    "p99(measurements.custom)",
                    "p99(measurements.another.custom)",
                ],
                "query": "",
                **self.additional_params,
            },
        )

        assert response.status_code == 200
        meta = response.data["p95(measurements.custom)"]["meta"]
        assert meta["fields"] == {
            "time": "date",
            "p95_measurements_custom": "size",
            "p99_measurements_custom": "size",
            "p99_measurements_another_custom": "size",
        }
        assert meta["units"] == {
            "time": None,
            "p95_measurements_custom": "kibibyte",
            "p99_measurements_custom": "kibibyte",
            "p99_measurements_another_custom": "pebibyte",
        }
        assert meta == response.data["p99(measurements.custom)"]["meta"]
        assert meta == response.data["p99(measurements.another.custom)"]["meta"]

    def test_no_top_events_with_project_field(self) -> None:
        project = self.create_project()
        response = self.do_request(
            data={
                # make sure to query the project with 0 events
                "project": project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "yAxis": "count()",
                "orderby": ["-count()"],
                "field": ["count()", "project"],
                "topEvents": 5,
                "dataset": "metrics",
                **self.additional_params,
            },
        )

        assert response.status_code == 200, response.content
        # When there are no top events, we do not return an empty dict.
        # Instead, we return a single zero-filled series for an empty graph.
        data = response.data["data"]
        assert [attrs for time, attrs in data] == [[{"count": 0}], [{"count": 0}]]

    def test_top_events_with_transaction(self) -> None:
        transaction_spec = [("foo", 100), ("bar", 200), ("baz", 300)]
        for offset in range(5):
            for transaction, duration in transaction_spec:
                self.store_transaction_metric(
                    duration,
                    tags={"transaction": f"{transaction}_transaction"},
                    timestamp=self.day_ago + timedelta(hours=offset, minutes=30),
                )

        response = self.do_request(
            data={
                # make sure to query the project with 0 events
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=5),
                "interval": "1h",
                "yAxis": "p75(transaction.duration)",
                "orderby": ["-p75(transaction.duration)"],
                "field": ["p75(transaction.duration)", "transaction"],
                "topEvents": 5,
                "dataset": "metrics",
                **self.additional_params,
            },
        )

        assert response.status_code == 200, response.content
        for position, (transaction, duration) in enumerate(transaction_spec):
            data = response.data[f"{transaction}_transaction"]
            chart_data = data["data"]
            assert data["order"] == 2 - position
            assert [attrs for time, attrs in chart_data] == [[{"count": duration}]] * 5

    def test_top_events_with_project(self) -> None:
        self.store_transaction_metric(
            100,
            timestamp=self.day_ago + timedelta(hours=1, minutes=30),
        )

        response = self.do_request(
            data={
                # make sure to query the project with 0 events
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=5),
                "interval": "1h",
                "yAxis": "p75(transaction.duration)",
                "orderby": ["-p75(transaction.duration)"],
                "field": ["p75(transaction.duration)", "project"],
                "topEvents": 5,
                "dataset": "metrics",
                **self.additional_params,
            },
        )

        assert response.status_code == 200, response.content
        data = response.data[f"{self.project.slug}"]
        assert data["order"] == 0

    def test_split_decision_for_errors_widget(self) -> None:
        error_data = load_data("python", timestamp=before_now(minutes=1))
        self.store_event(
            data={
                **error_data,
                "exception": {"values": [{"type": "blah", "data": {"values": []}}]},
            },
            project_id=self.project.id,
        )
        _, widget, __ = create_widget(
            ["count()", "error.type"], "error.type:blah", self.project, discover_widget_split=None
        )

        response = self.do_request(
            {
                "field": ["count()", "error.type"],
                "query": "error.type:blah",
                "dataset": "metricsEnhanced",
                "per_page": 50,
                "dashboardWidgetId": widget.id,
            }
        )

        assert response.status_code == 200, response.content
        assert response.data.get("meta").get(
            "discoverSplitDecision"
        ) is DashboardWidgetTypes.get_type_name(DashboardWidgetTypes.ERROR_EVENTS)

        widget.refresh_from_db()
        assert widget.discover_widget_split == DashboardWidgetTypes.ERROR_EVENTS
        assert widget.dataset_source == DatasetSourcesTypes.INFERRED.value

    def test_split_decision_for_transactions_widget(self) -> None:
        self.store_transaction_metric(
            100,
            timestamp=self.day_ago + timedelta(hours=1, minutes=30),
        )
        _, widget, __ = create_widget(
            ["count()", "transaction.name"], "", self.project, discover_widget_split=None
        )

        assert widget.discover_widget_split is None

        response = self.do_request(
            {
                "field": ["count()", "transaction.name"],
                "query": "",
                "dataset": "metricsEnhanced",
                "per_page": 50,
                "dashboardWidgetId": widget.id,
            }
        )

        assert response.status_code == 200, response.content
        assert response.data.get("meta").get(
            "discoverSplitDecision"
        ) is DashboardWidgetTypes.get_type_name(DashboardWidgetTypes.TRANSACTION_LIKE)

        widget.refresh_from_db()
        assert widget.discover_widget_split == DashboardWidgetTypes.TRANSACTION_LIKE
        assert widget.dataset_source == DatasetSourcesTypes.INFERRED.value

    def test_split_decision_for_top_events_errors_widget(self) -> None:
        error_data = load_data("python", timestamp=before_now(minutes=1))
        self.store_event(
            data={
                **error_data,
                "exception": {"values": [{"type": "test_error", "data": {"values": []}}]},
            },
            project_id=self.project.id,
        )
        _, widget, __ = create_widget(
            ["count()", "error.type"],
            "error.type:test_error",
            self.project,
            discover_widget_split=None,
        )

        response = self.do_request(
            {
                "field": ["count()", "error.type"],
                "query": "error.type:test_error",
                "dataset": "metricsEnhanced",
                "per_page": 50,
                "dashboardWidgetId": widget.id,
                "topEvents": 5,
            }
        )

        assert response.status_code == 200, response.content

        # Only a singular result for the test_error event
        assert len(response.data) == 1

        # Results are grouped by the error type
        assert response.data.get("[test_error]").get("meta").get(
            "discoverSplitDecision"
        ) is DashboardWidgetTypes.get_type_name(DashboardWidgetTypes.ERROR_EVENTS)

        widget.refresh_from_db()
        assert widget.discover_widget_split == DashboardWidgetTypes.ERROR_EVENTS
        assert widget.dataset_source == DatasetSourcesTypes.INFERRED.value

    def test_split_decision_for_top_events_transactions_widget(self) -> None:
        self.store_transaction_metric(
            100,
            timestamp=self.day_ago + timedelta(hours=1, minutes=30),
            tags={"transaction": "foo_transaction"},
        )
        _, widget, __ = create_widget(
            ["count()", "transaction"], "", self.project, discover_widget_split=None
        )

        assert widget.discover_widget_split is None

        response = self.do_request(
            {
                "field": ["count()", "transaction"],
                "query": "",
                "dataset": "metricsEnhanced",
                "per_page": 50,
                "dashboardWidgetId": widget.id,
                "topEvents": 5,
            }
        )

        assert response.status_code == 200, response.content

        # Only a singular result for the transaction
        assert len(response.data) == 1

        # Results are grouped by the transaction
        assert response.data.get("foo_transaction").get("meta").get(
            "discoverSplitDecision"
        ) is DashboardWidgetTypes.get_type_name(DashboardWidgetTypes.TRANSACTION_LIKE)

        widget.refresh_from_db()
        assert widget.discover_widget_split == DashboardWidgetTypes.TRANSACTION_LIKE
        assert widget.dataset_source == DatasetSourcesTypes.INFERRED.value

    def test_split_decision_for_ambiguous_widget_without_data(self) -> None:
        _, widget, __ = create_widget(
            ["count()", "transaction.name", "error.type"],
            "",
            self.project,
            discover_widget_split=None,
        )
        assert widget.discover_widget_split is None

        response = self.do_request(
            {
                "field": ["count()", "transaction.name", "error.type"],
                "query": "",
                "dataset": "metricsEnhanced",
                "per_page": 50,
                "dashboardWidgetId": widget.id,
            },
        )

        assert response.status_code == 200, response.content
        assert response.data.get("meta").get(
            "discoverSplitDecision"
        ) == DashboardWidgetTypes.get_type_name(DashboardWidgetTypes.ERROR_EVENTS)

        widget.refresh_from_db()
        assert widget.discover_widget_split == DashboardWidgetTypes.ERROR_EVENTS
        assert widget.dataset_source == DatasetSourcesTypes.FORCED.value

    def test_inp_percentile(self) -> None:
        for hour in range(6):
            timestamp = self.day_ago + timedelta(hours=hour, minutes=30)
            self.store_transaction_metric(
                111,
                metric="measurements.inp",
                timestamp=timestamp,
                use_case_id=UseCaseID.TRANSACTIONS,
            )

        response = self.do_request(
            data={
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=6),
                "interval": "1h",
                "yAxis": ["p75(measurements.inp)"],
                "project": self.project.id,
                "dataset": "metrics",
                **self.additional_params,
            },
        )
        assert response.status_code == 200, response.content
        data = response.data
        assert len(data["data"]) == 6
        assert data["isMetricsData"]
        assert data["meta"]["fields"]["p75_measurements_inp"] == "duration"
        for item in data["data"]:
            assert item[1][0]["count"] == 111

    def test_metrics_enhanced_defaults_to_transactions_with_feature_flag(self) -> None:
        # Store an error
        self.store_event(
            data={
                "event_id": "a" * 32,
                "message": "poof",
                "user": {"email": self.user.email},
                "timestamp": before_now(days=1, minutes=1).isoformat(),
                "tags": {"notMetrics": "this makes it not metrics"},
            },
            project_id=self.project.id,
        )

        # Store a transaction
        transaction_data = load_data("transaction")
        self.store_event(
            {
                **transaction_data,
                "tags": {"notMetrics": "this makes it not metrics"},
                "start_timestamp": before_now(days=1, minutes=1).isoformat(),
                "timestamp": before_now(days=1).isoformat(),
            },
            project_id=self.project.id,
        )
        features = {
            "organizations:discover-basic": True,
        }
        query = {
            "field": ["count()"],
            "query": 'notMetrics:"this makes it not metrics"',
            "statsPeriod": "1d",
            "interval": "1d",
            "dataset": "metricsEnhanced",
        }
        response = self.do_request(query, features=features)

        assert response.status_code == 200, response.content
        assert len(response.data["data"]) == 2

        # First bucket, where the transaction should be
        assert response.data["data"][0][1][0]["count"] == 1

    def test_metrics_enhanced_with_has_filter_falls_back_to_indexed_data(self) -> None:
        transaction_data = load_data("transaction")
        self.store_event(
            {
                **transaction_data,
                "tags": {"existingTag": "true"},
                "start_timestamp": before_now(days=1, minutes=1).isoformat(),
                "timestamp": before_now(days=1).isoformat(),
                "measurements": {"time_to_initial_display": {"value": 222}},
            },
            project_id=self.project.id,
        )

        for hour in range(6):
            timestamp = self.day_ago + timedelta(hours=hour, minutes=30)
            self.store_transaction_metric(
                111,
                metric="measurements.time_to_initial_display",
                timestamp=timestamp,
                tags={"existingTag": "true"},
            )
        query = {
            "field": ["avg(measurements.time_to_initial_display)"],
            "query": "has:existingTag !has:nonExistingTag",
            "statsPeriod": "1d",
            "interval": "1d",
            "dataset": "metricsEnhanced",
            "yAxis": ["avg(measurements.time_to_initial_display)"],
        }
        response = self.do_request(query, **self.additional_params)

        assert response.status_code == 200, response.content
        assert len(response.data["data"]) == 2

        # The request fell back to indexed data because the tag doesn't exist in the metrics indexer
        assert response.data["meta"]["isMetricsData"] is False

        # First bucket, where the transaction should be
        assert response.data["data"][0][1][0]["count"] == 222

    def test_top_events_with_metrics_enhanced_with_has_filter_falls_back_to_indexed_data(
        self,
    ) -> None:
        transaction_data = load_data("transaction")
        self.store_event(
            {
                **transaction_data,
                "tags": {"existingTag": "true"},
                "start_timestamp": before_now(days=1, minutes=1).isoformat(),
                "timestamp": before_now(days=1).isoformat(),
                "measurements": {"time_to_initial_display": {"value": 222}},
                "transaction": "foo_transaction",
            },
            project_id=self.project.id,
        )

        for hour in range(6):
            timestamp = self.day_ago + timedelta(hours=hour, minutes=30)
            self.store_transaction_metric(
                111,
                metric="measurements.time_to_initial_display",
                timestamp=timestamp,
                tags={"existingTag": "true"},
            )
        query = {
            "field": ["avg(measurements.time_to_initial_display)", "transaction"],
            "query": "has:existingTag !has:nonExistingTag",
            "statsPeriod": "1d",
            "interval": "1d",
            "dataset": "metricsEnhanced",
            "yAxis": ["avg(measurements.time_to_initial_display)"],
            "topEvents": 1,
        }
        response = self.do_request(query, **self.additional_params)

        assert response.status_code == 200, response.content
        assert len(response.data["foo_transaction"]["data"]) == 2

        data_series = response.data["foo_transaction"]

        # The request fell back to indexed data because the tag doesn't exist in the metrics indexer
        assert data_series["meta"]["isMetricsData"] is False

        # First bucket, where the transaction should be
        assert data_series["data"][0][1][0]["count"] == 222


class OrganizationEventsStatsMetricsEnhancedPerformanceEndpointTestWithOnDemandWidgets(
    MetricsEnhancedPerformanceTestCase
):
    endpoint = "sentry-api-0-organization-events-stats"

    def setUp(self) -> None:
        super().setUp()
        self.login_as(user=self.user)
        self.day_ago = before_now(days=1).replace(hour=10, minute=0, second=0, microsecond=0)
        self.DEFAULT_METRIC_TIMESTAMP = self.day_ago
        Environment.get_or_create(self.project, "production")

        self.url = reverse(
            "sentry-api-0-organization-events-stats",
            kwargs={"organization_id_or_slug": self.project.organization.slug},
        )
        self.features = {
            "organizations:on-demand-metrics-extraction-widgets": True,
            "organizations:on-demand-metrics-extraction": True,
        }

    def test_top_events_wrong_on_demand_type(self) -> None:
        query = "transaction.duration:>=100"
        yAxis = ["count()", "count_web_vitals(measurements.lcp, good)"]
        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": ["-count()"],
                "environment": "production",
                "query": query,
                "yAxis": yAxis,
                "field": [
                    "count()",
                ],
                "topEvents": 5,
                "dataset": "metrics",
                "useOnDemandMetrics": "true",
                "onDemandType": "not_real",
            },
        )

        assert response.status_code == 400, response.content

    def test_top_events_works_without_on_demand_type(self) -> None:
        query = "transaction.duration:>=100"
        yAxis = ["count()", "count_web_vitals(measurements.lcp, good)"]
        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": ["-count()"],
                "environment": "production",
                "query": query,
                "yAxis": yAxis,
                "field": [
                    "count()",
                ],
                "topEvents": 5,
                "dataset": "metrics",
                "useOnDemandMetrics": "true",
            },
        )

        assert response.status_code == 200, response.content

    def test_top_events_with_transaction_on_demand(self) -> None:
        field = "count()"
        field_two = "count_web_vitals(measurements.lcp, good)"
        groupbys = ["customtag1", "customtag2"]
        query = "transaction.duration:>=100"
        spec = OnDemandMetricSpec(
            field=field, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )
        spec_two = OnDemandMetricSpec(
            field=field_two, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )

        for hour in range(0, 5):
            self.store_on_demand_metric(
                hour * 62 * 24,
                spec=spec,
                additional_tags={
                    "customtag1": "foo",
                    "customtag2": "red",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                hour * 60 * 24,
                spec=spec_two,
                additional_tags={
                    "customtag1": "bar",
                    "customtag2": "blue",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )

        yAxis = ["count()", "count_web_vitals(measurements.lcp, good)"]

        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": ["-count()"],
                "environment": "production",
                "query": query,
                "yAxis": yAxis,
                "field": [
                    "count()",
                    "count_web_vitals(measurements.lcp, good)",
                    "customtag1",
                    "customtag2",
                ],
                "topEvents": 5,
                "dataset": "metricsEnhanced",
                "useOnDemandMetrics": "true",
                "onDemandType": "dynamic_query",
            },
        )

        assert response.status_code == 200, response.content

        groups = [
            ("foo,red", "count()", 0.0, 1488.0),
            ("foo,red", "count_web_vitals(measurements.lcp, good)", 0.0, 0.0),
            ("bar,blue", "count()", 0.0, 0.0),
            ("bar,blue", "count_web_vitals(measurements.lcp, good)", 0.0, 1440.0),
        ]
        assert len(response.data.keys()) == 2
        for group_count in groups:
            group, agg, row1, row2 = group_count
            row_data = response.data[group][agg]["data"][:2]
            assert [attrs for _, attrs in row_data] == [[{"count": row1}], [{"count": row2}]]

            assert response.data[group][agg]["meta"]["isMetricsExtractedData"]
            assert response.data[group]["isMetricsExtractedData"]

    def test_top_events_with_transaction_on_demand_and_no_environment(self) -> None:
        field = "count()"
        field_two = "count_web_vitals(measurements.lcp, good)"
        groupbys = ["customtag1", "customtag2"]
        query = "transaction.duration:>=100"
        spec = OnDemandMetricSpec(
            field=field, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )
        spec_two = OnDemandMetricSpec(
            field=field_two, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )

        for hour in range(0, 5):
            self.store_on_demand_metric(
                hour * 62 * 24,
                spec=spec,
                additional_tags={
                    "customtag1": "foo",
                    "customtag2": "red",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                hour * 60 * 24,
                spec=spec_two,
                additional_tags={
                    "customtag1": "bar",
                    "customtag2": "blue",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )

        yAxis = ["count()", "count_web_vitals(measurements.lcp, good)"]

        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": ["-count()"],
                "query": query,
                "yAxis": yAxis,
                "field": [
                    "count()",
                    "count_web_vitals(measurements.lcp, good)",
                    "customtag1",
                    "customtag2",
                ],
                "topEvents": 5,
                "dataset": "metricsEnhanced",
                "useOnDemandMetrics": "true",
                "onDemandType": "dynamic_query",
            },
        )

        assert response.status_code == 200, response.content

        groups = [
            ("foo,red", "count()", 0.0, 1488.0),
            ("foo,red", "count_web_vitals(measurements.lcp, good)", 0.0, 0.0),
            ("bar,blue", "count()", 0.0, 0.0),
            ("bar,blue", "count_web_vitals(measurements.lcp, good)", 0.0, 1440.0),
        ]
        assert len(response.data.keys()) == 2
        for group_count in groups:
            group, agg, row1, row2 = group_count
            row_data = response.data[group][agg]["data"][:2]
            assert [attrs for time, attrs in row_data] == [[{"count": row1}], [{"count": row2}]]

            assert response.data[group][agg]["meta"]["isMetricsExtractedData"]
            assert response.data[group]["isMetricsExtractedData"]

    def test_top_events_with_transaction_on_demand_passing_widget_id_unsaved_transaction_only(
        self,
    ) -> None:
        field = "count()"
        field_two = "count_web_vitals(measurements.lcp, good)"
        groupbys = ["customtag1", "customtag2"]
        query = "transaction.duration:>=100"
        spec = OnDemandMetricSpec(
            field=field, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )
        spec_two = OnDemandMetricSpec(
            field=field_two, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )

        _, widget, __ = create_widget(
            ["count()"],
            "",
            self.project,
            discover_widget_split=None,
        )

        for hour in range(0, 2):
            self.store_on_demand_metric(
                hour * 62 * 24,
                spec=spec,
                additional_tags={
                    "customtag1": "foo",
                    "customtag2": "red",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                hour * 60 * 24,
                spec=spec_two,
                additional_tags={
                    "customtag1": "bar",
                    "customtag2": "blue",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )

        yAxis = [field, field_two]

        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": ["-count()"],
                "query": query,
                "yAxis": yAxis,
                "field": yAxis + groupbys,
                "topEvents": 5,
                "dataset": "metricsEnhanced",
                "useOnDemandMetrics": "true",
                "onDemandType": "dynamic_query",
                "dashboardWidgetId": widget.id,
            },
        )
        saved_widget = DashboardWidget.objects.get(id=widget.id)
        assert saved_widget.discover_widget_split == DashboardWidgetTypes.TRANSACTION_LIKE

        assert response.status_code == 200, response.content
        # Fell back to discover data which is empty for this test (empty group of '').
        assert len(response.data.keys()) == 2
        assert bool(response.data["foo,red"])
        assert bool(response.data["bar,blue"])

    def test_top_events_with_transaction_on_demand_passing_widget_id_unsaved_error(
        self,
    ):
        self.project = self.create_project(organization=self.organization)
        Environment.get_or_create(self.project, "production")
        field = "count()"
        field_two = "count()"
        groupbys = ["customtag1", "customtag2"]
        query = "query.dataset:foo"

        _, widget, __ = create_widget(
            ["count()"],
            "",
            self.project,
            discover_widget_split=None,
        )

        self.store_event(
            data={
                "event_id": "a" * 32,
                "message": "very bad",
                "type": "error",
                "start_timestamp": (self.day_ago + timedelta(hours=1)).isoformat(),
                "timestamp": (self.day_ago + timedelta(hours=1)).isoformat(),
                "tags": {"customtag1": "error_value", "query.dataset": "foo"},
            },
            project_id=self.project.id,
        )
        self.store_event(
            data={
                "event_id": "b" * 32,
                "message": "very bad 2",
                "type": "error",
                "start_timestamp": (self.day_ago + timedelta(hours=1)).isoformat(),
                "timestamp": (self.day_ago + timedelta(hours=1)).isoformat(),
                "tags": {"customtag1": "error_value2", "query.dataset": "foo"},
            },
            project_id=self.project.id,
        )

        yAxis = ["count()"]

        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": ["-count()"],
                "query": query,
                "yAxis": yAxis,
                "field": [field, field_two] + groupbys,
                "topEvents": 5,
                "dataset": "metricsEnhanced",
                "useOnDemandMetrics": "true",
                "onDemandType": "dynamic_query",
                "dashboardWidgetId": widget.id,
            },
        )
        saved_widget = DashboardWidget.objects.get(id=widget.id)
        assert saved_widget.discover_widget_split == DashboardWidgetTypes.ERROR_EVENTS

        assert response.status_code == 200, response.content
        # Fell back to discover data which is empty for this test (empty group of '').
        assert len(response.data.keys()) == 2
        assert bool(response.data["error_value,"])
        assert bool(response.data["error_value2,"])

    def test_top_events_with_transaction_on_demand_passing_widget_id_unsaved_discover(self) -> None:
        self.project = self.create_project(organization=self.organization)
        Environment.get_or_create(self.project, "production")
        field = "count()"
        field_two = "count()"
        groupbys = ["customtag1", "customtag2"]
        query = "query.dataset:foo"
        spec = OnDemandMetricSpec(
            field=field, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )
        spec_two = OnDemandMetricSpec(
            field=field_two, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )

        _, widget, __ = create_widget(
            ["count()"],
            "",
            self.project,
            discover_widget_split=None,
        )

        self.store_event(
            data={
                "event_id": "a" * 32,
                "message": "very bad",
                "type": "error",
                "timestamp": (self.day_ago + timedelta(hours=1)).isoformat(),
                "tags": {"customtag1": "error_value", "query.dataset": "foo"},
            },
            project_id=self.project.id,
        )

        transaction = load_data("transaction")
        transaction["timestamp"] = (self.day_ago + timedelta(hours=1)).isoformat()
        transaction["start_timestamp"] = (self.day_ago + timedelta(hours=1)).isoformat()
        transaction["tags"] = {"customtag1": "transaction_value", "query.dataset": "foo"}

        self.store_event(
            data=transaction,
            project_id=self.project.id,
        )

        for hour in range(0, 5):
            self.store_on_demand_metric(
                hour * 62 * 24,
                spec=spec,
                additional_tags={
                    "customtag1": "foo",
                    "customtag2": "red",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                hour * 60 * 24,
                spec=spec_two,
                additional_tags={
                    "customtag1": "bar",
                    "customtag2": "blue",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )

        yAxis = ["count()"]

        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": ["-count()"],
                "query": query,
                "yAxis": yAxis,
                "field": [field, field_two, "customtag1", "customtag2"],
                "topEvents": 5,
                "dataset": "metricsEnhanced",
                "useOnDemandMetrics": "true",
                "onDemandType": "dynamic_query",
                "dashboardWidgetId": widget.id,
            },
        )

        saved_widget = DashboardWidget.objects.get(id=widget.id)
        assert (
            saved_widget.discover_widget_split == DashboardWidgetTypes.ERROR_EVENTS
        )  # we default to errors

        assert response.status_code == 200, response.content

        assert response.status_code == 200, response.content
        # Fell back to discover data which is empty for this test (empty group of '').
        assert len(response.data.keys()) == 1
        assert bool(response.data["error_value,"])

    def test_top_events_with_transaction_on_demand_passing_widget_id_saved(self) -> None:
        field = "count()"
        field_two = "count_web_vitals(measurements.lcp, good)"
        groupbys = ["customtag1", "customtag2"]
        query = "transaction.duration:>=100"
        spec = OnDemandMetricSpec(
            field=field, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )
        spec_two = OnDemandMetricSpec(
            field=field_two, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )

        _, widget, __ = create_widget(
            ["count()"],
            "",
            self.project,
            discover_widget_split=DashboardWidgetTypes.TRANSACTION_LIKE,  # Transactions like uses on-demand
        )

        for hour in range(0, 5):
            self.store_on_demand_metric(
                hour * 62 * 24,
                spec=spec,
                additional_tags={
                    "customtag1": "foo",
                    "customtag2": "red",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                hour * 60 * 24,
                spec=spec_two,
                additional_tags={
                    "customtag1": "bar",
                    "customtag2": "blue",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )

        yAxis = ["count()", "count_web_vitals(measurements.lcp, good)"]

        with mock.patch.object(widget, "save") as mock_widget_save:
            response = self.do_request(
                data={
                    "project": self.project.id,
                    "start": self.day_ago,
                    "end": self.day_ago + timedelta(hours=2),
                    "interval": "1h",
                    "orderby": ["-count()"],
                    "query": query,
                    "yAxis": yAxis,
                    "field": [
                        "count()",
                        "count_web_vitals(measurements.lcp, good)",
                        "customtag1",
                        "customtag2",
                    ],
                    "topEvents": 5,
                    "dataset": "metricsEnhanced",
                    "useOnDemandMetrics": "true",
                    "onDemandType": "dynamic_query",
                    "dashboardWidgetId": widget.id,
                },
            )
            assert bool(mock_widget_save.assert_not_called)

        assert response.status_code == 200, response.content

        groups = [
            ("foo,red", "count()", 0.0, 1488.0),
            ("foo,red", "count_web_vitals(measurements.lcp, good)", 0.0, 0.0),
            ("bar,blue", "count()", 0.0, 0.0),
            ("bar,blue", "count_web_vitals(measurements.lcp, good)", 0.0, 1440.0),
        ]
        assert len(response.data.keys()) == 2
        for group_count in groups:
            group, agg, row1, row2 = group_count
            row_data = response.data[group][agg]["data"][:2]
            assert [attrs for time, attrs in row_data] == [[{"count": row1}], [{"count": row2}]]

            assert response.data[group][agg]["meta"]["isMetricsExtractedData"]
            assert response.data[group]["isMetricsExtractedData"]

    def test_timeseries_on_demand_with_multiple_percentiles(self) -> None:
        field = "p75(measurements.fcp)"
        field_two = "p75(measurements.lcp)"
        query = "transaction.duration:>=100"
        spec = OnDemandMetricSpec(field=field, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY)
        spec_two = OnDemandMetricSpec(
            field=field_two, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )

        assert (
            spec._query_str_for_hash
            == "event.measurements.fcp.value;{'name': 'event.duration', 'op': 'gte', 'value': 100.0}"
        )
        assert (
            spec_two._query_str_for_hash
            == "event.measurements.lcp.value;{'name': 'event.duration', 'op': 'gte', 'value': 100.0}"
        )

        for count in range(0, 4):
            self.store_on_demand_metric(
                count * 100,
                spec=spec,
                timestamp=self.day_ago + timedelta(hours=1),
            )
            self.store_on_demand_metric(
                count * 200.0,
                spec=spec_two,
                timestamp=self.day_ago + timedelta(hours=1),
            )

        yAxis = [field, field_two]

        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": [field],
                "query": query,
                "yAxis": yAxis,
                "dataset": "metricsEnhanced",
                "useOnDemandMetrics": "true",
                "onDemandType": "dynamic_query",
            },
        )

        assert response.status_code == 200, response.content

        assert response.data["p75(measurements.fcp)"]["meta"]["isMetricsExtractedData"]
        assert response.data["p75(measurements.lcp)"]["meta"]["isMetricsData"]
        assert [attrs for time, attrs in response.data["p75(measurements.fcp)"]["data"]] == [
            [{"count": 0}],
            [{"count": 225.0}],
        ]
        assert response.data["p75(measurements.lcp)"]["meta"]["isMetricsExtractedData"]
        assert response.data["p75(measurements.lcp)"]["meta"]["isMetricsData"]
        assert [attrs for time, attrs in response.data["p75(measurements.lcp)"]["data"]] == [
            [{"count": 0}],
            [{"count": 450.0}],
        ]

    def test_apdex_issue(self) -> None:
        field = "apdex(300)"
        groupbys = ["group_tag"]
        query = "transaction.duration:>=100"
        spec = OnDemandMetricSpec(
            field=field,
            groupbys=groupbys,
            query=query,
            spec_type=MetricSpecType.DYNAMIC_QUERY,
        )

        for hour in range(0, 5):
            self.store_on_demand_metric(
                1,
                spec=spec,
                additional_tags={
                    "group_tag": "group_one",
                    "environment": "production",
                    "satisfaction": "tolerable",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                1,
                spec=spec,
                additional_tags={
                    "group_tag": "group_two",
                    "environment": "production",
                    "satisfaction": "satisfactory",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )

        response = self.do_request(
            data={
                "dataset": "metricsEnhanced",
                "environment": "production",
                "excludeOther": 1,
                "field": [field, "group_tag"],
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": f"-{field}",
                "partial": 1,
                "project": self.project.id,
                "query": query,
                "topEvents": 5,
                "yAxis": field,
                "onDemandType": "dynamic_query",
                "useOnDemandMetrics": "true",
            },
        )

        assert response.status_code == 200, response.content
        assert response.data["group_one"]["meta"]["isMetricsExtractedData"] is True
        assert [attrs for time, attrs in response.data["group_one"]["data"]] == [
            [{"count": 0.5}],
            [{"count": 0.5}],
        ]

    def test_glob_http_referer_on_demand(self) -> None:
        agg = "count()"
        network_id_tag = "networkId"
        url = "https://sentry.io"
        query = f'http.url:{url}/*/foo/bar/* http.referer:"{url}/*/bar/*" event.type:transaction'
        spec = OnDemandMetricSpec(
            field=agg,
            groupbys=[network_id_tag],
            query=query,
            spec_type=MetricSpecType.DYNAMIC_QUERY,
        )
        assert spec.to_metric_spec(self.project) == {
            "category": "transaction",
            "mri": "c:transactions/on_demand@none",
            "field": None,
            "tags": [
                {"key": "query_hash", "value": "ac241f56"},
                {"key": "networkId", "field": "event.tags.networkId"},
                {"key": "environment", "field": "event.environment"},
            ],
            "condition": {
                "op": "and",
                "inner": [
                    {
                        "op": "glob",
                        "name": "event.request.url",
                        "value": ["https://sentry.io/*/foo/bar/*"],
                    },
                    {
                        "op": "glob",
                        "name": "event.request.headers.Referer",
                        "value": ["https://sentry.io/*/bar/*"],
                    },
                ],
            },
        }

        for hour in range(0, 5):
            self.store_on_demand_metric(
                1,
                spec=spec,
                additional_tags={network_id_tag: "1234"},
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                1,
                spec=spec,
                additional_tags={network_id_tag: "5678"},
                timestamp=self.day_ago + timedelta(hours=hour),
            )

        response = self.do_request(
            data={
                "dataset": "metricsEnhanced",
                "field": [network_id_tag, agg],
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=5),
                "onDemandType": "dynamic_query",
                "orderby": f"-{agg}",
                "interval": "1d",
                "partial": 1,
                "query": query,
                "referrer": "api.dashboards.widget.bar-chart",
                "project": self.project.id,
                "topEvents": 2,
                "useOnDemandMetrics": "true",
                "yAxis": agg,
            },
        )

        assert response.status_code == 200, response.content
        for datum in response.data.values():
            assert datum["meta"] == {
                "dataScanned": "full",
                "dataset": "metricsEnhanced",
                "datasetReason": "unchanged",
                "fields": {"count": "integer", "networkId": "string", "time": "date"},
                "isMetricsData": False,
                "isMetricsExtractedData": True,
                "tips": {},
                "units": {"count": None, "networkId": None, "time": None},
            }

    def _test_is_metrics_extracted_data(
        self, params: dict[str, Any], expected_on_demand_query: bool, dataset: str
    ) -> None:
        spec = OnDemandMetricSpec(
            field="count()",
            query="transaction.duration:>1s",
            spec_type=MetricSpecType.DYNAMIC_QUERY,
        )

        self.store_on_demand_metric(1, spec=spec)
        response = self.do_request(params)

        assert response.status_code == 200, response.content
        meta = response.data["meta"]
        # This is the main thing we want to test for
        assert meta.get("isMetricsExtractedData", False) is expected_on_demand_query
        assert meta["dataset"] == dataset

        return meta

    def test_is_metrics_extracted_data_is_included(self) -> None:
        self._test_is_metrics_extracted_data(
            {
                "dataset": "metricsEnhanced",
                "query": "transaction.duration:>=91",
                "useOnDemandMetrics": "true",
                "yAxis": "count()",
            },
            expected_on_demand_query=True,
            dataset="metricsEnhanced",
        )

    def test_on_demand_epm_no_query(self) -> None:
        params = {
            "dataset": "metricsEnhanced",
            "environment": "production",
            "onDemandType": "dynamic_query",
            "project": self.project.id,
            "query": "",
            "statsPeriod": "1h",
            "useOnDemandMetrics": "true",
            "yAxis": ["epm()"],
        }
        response = self.do_request(params)

        assert response.status_code == 200, response.content
        assert response.data["meta"] == {
            "dataScanned": "full",
            "fields": {"time": "date", "epm": "rate"},
            "units": {"time": None, "epm": None},
            "isMetricsData": True,
            "isMetricsExtractedData": False,
            "tips": {},
            "datasetReason": "unchanged",
            "dataset": "metricsEnhanced",
        }

    def test_group_by_transaction(self) -> None:
        field = "count()"
        groupbys = ["transaction"]
        query = "transaction.duration:>=100"
        spec = OnDemandMetricSpec(
            field=field,
            groupbys=groupbys,
            query=query,
            spec_type=MetricSpecType.DYNAMIC_QUERY,
        )

        for hour in range(0, 2):
            self.store_on_demand_metric(
                (hour + 1) * 5,
                spec=spec,
                additional_tags={
                    "transaction": "/performance",
                    "environment": "production",
                },
                timestamp=self.day_ago + timedelta(hours=hour),
            )

        response = self.do_request(
            data={
                "dataset": "metricsEnhanced",
                "environment": "production",
                "excludeOther": 1,
                "field": [field, "transaction"],
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": f"-{field}",
                "partial": 1,
                "project": self.project.id,
                "query": query,
                "topEvents": 5,
                "yAxis": field,
                "onDemandType": "dynamic_query",
                "useOnDemandMetrics": "true",
            },
        )

        assert response.status_code == 200, response.content
        assert response.data["/performance"]["meta"]["isMetricsExtractedData"] is True
        assert [attrs for time, attrs in response.data["/performance"]["data"]] == [
            [{"count": 5.0}],
            [{"count": 10.0}],
        ]

    def _setup_orderby_tests(self, query):
        count_spec = OnDemandMetricSpec(
            field="count()",
            groupbys=["networkId"],
            query=query,
            spec_type=MetricSpecType.DYNAMIC_QUERY,
        )
        p95_spec = OnDemandMetricSpec(
            field="p95(transaction.duration)",
            groupbys=["networkId"],
            query=query,
            spec_type=MetricSpecType.DYNAMIC_QUERY,
        )

        for hour in range(0, 5):
            self.store_on_demand_metric(
                1,
                spec=count_spec,
                additional_tags={"networkId": "1234"},
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                100,
                spec=p95_spec,
                additional_tags={"networkId": "1234"},
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            self.store_on_demand_metric(
                200,
                spec=p95_spec,
                additional_tags={"networkId": "5678"},
                timestamp=self.day_ago + timedelta(hours=hour),
            )
            # Store twice as many 5678 so orderby puts it later
            self.store_on_demand_metric(
                2,
                spec=count_spec,
                additional_tags={"networkId": "5678"},
                timestamp=self.day_ago + timedelta(hours=hour),
            )

    def test_order_by_aggregate_top_events_desc(self) -> None:
        url = "https://sentry.io"
        query = f'http.url:{url}/*/foo/bar/* http.referer:"{url}/*/bar/*" event.type:transaction'
        self._setup_orderby_tests(query)
        response = self.do_request(
            data={
                "dataset": "metricsEnhanced",
                "field": ["networkId", "count()"],
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=5),
                "onDemandType": "dynamic_query",
                "orderby": "-count()",
                "interval": "1d",
                "partial": 1,
                "query": query,
                "referrer": "api.dashboards.widget.bar-chart",
                "project": self.project.id,
                "topEvents": 2,
                "useOnDemandMetrics": "true",
                "yAxis": "count()",
            },
        )

        assert response.status_code == 200, response.content
        assert len(response.data) == 3
        data1 = response.data["5678"]
        assert data1["order"] == 0
        assert data1["data"][0][1][0]["count"] == 10
        data2 = response.data["1234"]
        assert data2["order"] == 1
        assert data2["data"][0][1][0]["count"] == 5
        for datum in response.data.values():
            assert datum["meta"] == {
                "dataScanned": "full",
                "dataset": "metricsEnhanced",
                "datasetReason": "unchanged",
                "fields": {"count": "integer", "networkId": "string", "time": "date"},
                "isMetricsData": False,
                "isMetricsExtractedData": True,
                "tips": {},
                "units": {"count": None, "networkId": None, "time": None},
            }

    def test_order_by_aggregate_top_events_asc(self) -> None:
        url = "https://sentry.io"
        query = f'http.url:{url}/*/foo/bar/* http.referer:"{url}/*/bar/*" event.type:transaction'
        self._setup_orderby_tests(query)
        response = self.do_request(
            data={
                "dataset": "metricsEnhanced",
                "field": ["networkId", "count()"],
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=5),
                "onDemandType": "dynamic_query",
                "orderby": "count()",
                "interval": "1d",
                "partial": 1,
                "query": query,
                "referrer": "api.dashboards.widget.bar-chart",
                "project": self.project.id,
                "topEvents": 2,
                "useOnDemandMetrics": "true",
                "yAxis": "count()",
            },
        )

        assert response.status_code == 200, response.content
        assert len(response.data) == 3
        data1 = response.data["1234"]
        assert data1["order"] == 0
        assert data1["data"][0][1][0]["count"] == 5
        data2 = response.data["5678"]
        assert data2["order"] == 1
        assert data2["data"][0][1][0]["count"] == 10
        for datum in response.data.values():
            assert datum["meta"] == {
                "dataScanned": "full",
                "dataset": "metricsEnhanced",
                "datasetReason": "unchanged",
                "fields": {"count": "integer", "networkId": "string", "time": "date"},
                "isMetricsData": False,
                "isMetricsExtractedData": True,
                "tips": {},
                "units": {"count": None, "networkId": None, "time": None},
            }

    def test_order_by_aggregate_top_events_graph_different_aggregate(self) -> None:
        url = "https://sentry.io"
        query = f'http.url:{url}/*/foo/bar/* http.referer:"{url}/*/bar/*" event.type:transaction'
        self._setup_orderby_tests(query)
        response = self.do_request(
            data={
                "dataset": "metricsEnhanced",
                "field": ["networkId", "count()"],
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=5),
                "onDemandType": "dynamic_query",
                "orderby": "count()",
                "interval": "1d",
                "partial": 1,
                "query": query,
                "referrer": "api.dashboards.widget.bar-chart",
                "project": self.project.id,
                "topEvents": 2,
                "useOnDemandMetrics": "true",
                "yAxis": "p95(transaction.duration)",
            },
        )

        assert response.status_code == 200, response.content
        assert len(response.data) == 3
        data1 = response.data["1234"]
        assert data1["order"] == 0
        assert data1["data"][0][1][0]["count"] == 100
        data2 = response.data["5678"]
        assert data2["order"] == 1
        assert data2["data"][0][1][0]["count"] == 200
        for datum in response.data.values():
            assert datum["meta"] == {
                "dataScanned": "full",
                "dataset": "metricsEnhanced",
                "datasetReason": "unchanged",
                "fields": {
                    "networkId": "string",
                    "p95_transaction_duration": "duration",
                    "time": "date",
                },
                "isMetricsData": False,
                "isMetricsExtractedData": True,
                "tips": {},
                "units": {
                    "networkId": None,
                    "p95_transaction_duration": "millisecond",
                    "time": None,
                },
            }

    def test_cannot_order_by_tag(self) -> None:
        url = "https://sentry.io"
        query = f'http.url:{url}/*/foo/bar/* http.referer:"{url}/*/bar/*" event.type:transaction'
        self._setup_orderby_tests(query)
        response = self.do_request(
            data={
                "dataset": "metrics",
                "field": ["networkId", "count()"],
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=5),
                "onDemandType": "dynamic_query",
                "orderby": "-networkId",
                "interval": "1d",
                "partial": 1,
                "query": query,
                "referrer": "api.dashboards.widget.bar-chart",
                "project": self.project.id,
                "topEvents": 2,
                "useOnDemandMetrics": "true",
                "yAxis": "count()",
            },
        )

        assert response.status_code == 400, response.content

    def test_order_by_two_aggregates(self) -> None:
        url = "https://sentry.io"
        query = f'http.url:{url}/*/foo/bar/* http.referer:"{url}/*/bar/*" event.type:transaction'
        self._setup_orderby_tests(query)
        response = self.do_request(
            data={
                "dataset": "metrics",
                "field": ["networkId", "count()", "p95(transaction.duration)"],
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=5),
                "onDemandType": "dynamic_query",
                "orderby": ["count()", "p95(transaction.duration)"],
                "interval": "1d",
                "partial": 1,
                "query": query,
                "referrer": "api.dashboards.widget.bar-chart",
                "project": self.project.id,
                "topEvents": 2,
                "useOnDemandMetrics": "true",
                "yAxis": "p95(transaction.duration)",
            },
        )

        assert response.status_code == 400, response.content

    def test_top_events_with_tag(self) -> None:
        query = "transaction.duration:>=100"
        yAxis = ["count()"]
        field = "count()"
        groupbys = ["some-field"]
        spec = OnDemandMetricSpec(
            field=field, groupbys=groupbys, query=query, spec_type=MetricSpecType.DYNAMIC_QUERY
        )
        self.store_on_demand_metric(
            1,
            spec=spec,
            additional_tags={
                "some-field": "bar",
                "environment": "production",
            },
            timestamp=self.day_ago,
        )
        response = self.do_request(
            data={
                "project": self.project.id,
                "start": self.day_ago,
                "end": self.day_ago + timedelta(hours=2),
                "interval": "1h",
                "orderby": ["-count()"],
                "environment": "production",
                "query": query,
                "yAxis": yAxis,
                "field": [
                    "some-field",
                    "count()",
                ],
                "topEvents": 5,
                "dataset": "metrics",
                "useOnDemandMetrics": "true",
            },
        )

        assert response.status_code == 200, response.content
